#ifndef NVLOGGER_H
#define NVLOGGER_H

#include <QDebug>
#include <NvInferRuntimeCommon.h>

class NvLogger : public nvinfer1::ILogger
{
public:
    NvLogger(nvinfer1::ILogger::Severity level);

public:
    void log(nvinfer1::ILogger::Severity severity, nvinfer1::AsciiChar const* msg) noexcept;

private:
    nvinfer1::ILogger::Severity level;
};

#endif // NVLOGGER_H
